package com.woniu.zsd.order.util;


import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;
import java.util.UUID;

@Component("orderOssUtil")
public class OssUtil {

    @Value("${aliyun.oss.host}")
    private String endpoint;

    @Value("${aliyun.oss.accessKey}")
    private String accessKeyId;

    @Value("${aliyun.oss.accessSecret}")
    private String accessKeySecret;

    @Value("${aliyun.oss.bucketName}")
    private String bucketName;

    /**
     * 上传文件到阿里云 OSS
     *
     * @param file 文件
     * @return 文件的访问 URL
     */
    public String uploadFile(MultipartFile file) {
        // 获取文件名并生成唯一文件名
        String originalFilename = file.getOriginalFilename();
        String fileName = UUID.randomUUID().toString() + "." + getFileExtension(originalFilename);

        // 创建 OSS 客户端
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        try {
            // 获取文件输入流
            InputStream inputStream = file.getInputStream();

            // 上传文件
            ossClient.putObject(bucketName, fileName, inputStream);

            // 移除 endpoint 中的协议头（http:// 或 https://）
            String cleanEndpoint = endpoint.replaceAll("^https?://", "");

            // 拼接文件访问 URL
            String fileUrl = "https://" + bucketName + "." + cleanEndpoint + "/" + fileName;
            return fileUrl;
        } catch (Exception e) {
            throw new RuntimeException("文件上传失败", e);
        } finally {
            // 关闭 OSS 客户端
            if (ossClient != null) {
                ossClient.shutdown();
            }
        }
    }

    /**
     * 获取文件扩展名
     *
     * @param fileName 文件名
     * @return 文件扩展名
     */
    private String getFileExtension(String fileName) {
        if (fileName == null || !fileName.contains(".")) {
            throw new IllegalArgumentException("文件名格式错误");
        }
        return fileName.substring(fileName.lastIndexOf(".") + 1);
    }
}